home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Demo / sockets / throughput.py < prev    next >
Encoding:
Python Source  |  2000-06-23  |  1.8 KB  |  94 lines
  1. #! /usr/bin/env python
  2.  
  3. # Test network throughput.
  4. #
  5. # Usage:
  6. # 1) on host_A: throughput -s [port]            # start a server
  7. # 2) on host_B: throughput -c  count host_A [port]    # start a client
  8. #
  9. # The server will service multiple clients until it is killed.
  10. #
  11. # The client performs one transfer of count*BUFSIZE bytes and
  12. # measures the time it takes (roundtrip!).
  13.  
  14.  
  15. import sys, time
  16. from socket import *
  17.  
  18. MY_PORT = 50000 + 42
  19.  
  20. BUFSIZE = 1024
  21.  
  22.  
  23. def main():
  24.     if len(sys.argv) < 2:
  25.         usage()
  26.     if sys.argv[1] == '-s':
  27.         server()
  28.     elif sys.argv[1] == '-c':
  29.         client()
  30.     else:
  31.         usage()
  32.  
  33.  
  34. def usage():
  35.     sys.stdout = sys.stderr
  36.     print 'Usage:    (on host_A) throughput -s [port]'
  37.     print 'and then: (on host_B) throughput -c count host_A [port]'
  38.     sys.exit(2)
  39.  
  40.  
  41. def server():
  42.     if len(sys.argv) > 2:
  43.         port = eval(sys.argv[2])
  44.     else:
  45.         port = MY_PORT
  46.     s = socket(AF_INET, SOCK_STREAM)
  47.     s.bind('', port)
  48.     s.listen(1)
  49.     print 'Server ready...'
  50.     while 1:
  51.         conn, (host, remoteport) = s.accept()
  52.         while 1:
  53.             data = conn.recv(BUFSIZE)
  54.             if not data:
  55.                 break
  56.             del data
  57.         conn.send('OK\n')
  58.         conn.close()
  59.         print 'Done with', host, 'port', remoteport
  60.  
  61.  
  62. def client():
  63.     if len(sys.argv) < 4:
  64.         usage()
  65.     count = int(eval(sys.argv[2]))
  66.     host = sys.argv[3]
  67.     if len(sys.argv) > 4:
  68.         port = eval(sys.argv[4])
  69.     else:
  70.         port = MY_PORT
  71.     testdata = 'x' * (BUFSIZE-1) + '\n'
  72.     t1 = time.time()
  73.     s = socket(AF_INET, SOCK_STREAM)
  74.     t2 = time.time()
  75.     s.connect(host, port)
  76.     t3 = time.time()
  77.     i = 0
  78.     while i < count:
  79.         i = i+1
  80.         s.send(testdata)
  81.     s.shutdown(1) # Send EOF
  82.     t4 = time.time()
  83.     data = s.recv(BUFSIZE)
  84.     t5 = time.time()
  85.     print data
  86.     print 'Raw timers:', t1, t2, t3, t4, t5
  87.     print 'Intervals:', t2-t1, t3-t2, t4-t3, t5-t4
  88.     print 'Total:', t5-t1
  89.     print 'Throughput:', round((BUFSIZE*count*0.001) / (t5-t1), 3),
  90.     print 'K/sec.'
  91.  
  92.  
  93. main()
  94.